Class (machine learning)

A category that a label can belong to. For example:1

  • In a binary classification model that detects spam, the two classes might be spam and not spam.
  • In a multi-class classification model that identifies dog breeds, the classes might be poodle, beagle, pug, and so on.

A classification model predicts a class. In contrast, a regression model predicts a number rather than a class.1

Footnotes

  1. developers.google.com/machine-learning/glossary#class 2

2024 © ak